export DATA_DIR=/data/sdv1/wuwanjie

python trainer.py \
    --model_name_or_path $DATA_DIR/transformers/chinese_roberta_wwm_ext \
    --train_path $DATA_DIR/datasets/CSTS/PKU-Paraphrase-Bank/paraphrase_unorder \
    --dev_path $DATA_DIR/datasets/CSTS/Chinese-STS-B/sts-b-dev.txt \
    --test_path $DATA_DIR/datasets/CSTS/Chinese-STS-B/sts-b-test.txt \
    --output_dir $DATA_DIR/outputs \
    --dropout 0.4 \
    --epochs 1 \
    --temperature 0.05 \
    --pooler last-avg \
    --batch_size 128 \
    --learning_rate 5e-5 \
    --max_seq_length 64 \
    --max_grad_norm 10.0 \
    --evaluation_per_epoch 5 \
    --mode unsup